/*
 * 这个版本是有问题的
 * √ 首先，在超时踢出的设置（每秒检查）与踢出时机（while末尾）
 * × 其次，在循环内直接删除迭代器指向的元素可能会有问题，所以尽量存储元素后再删除
 *
 * */

#include <sys/socket.h>     // 包含socket()、bind()、connect()等函数声明
#include <netinet/in.h>     // 包含AF_INET、SOCK_STREAM等常量定义
#include <arpa/inet.h>      // 包含inet_addr()等IP地址处理函数（如果需要）
#include <sys/epoll.h>      // 包含epoll()函数
#include <unistd.h>         // 包含close()函数
#include <time.h>           // 包含时间库
#include <fcntl.h>          // 提供 fcntl()函数及相关常量（F_GETFL、O_NONBLOCK等）

#include <cstdio>
#include <cstring>          // 提供std::strerror
#include <string>
#include <vector>
#include <algorithm>        // find_if
using std::string;
using std::vector;

#define MAX_EVENTS 10
#define BUFFER_SIZE 1024

struct Client               // 客户端信息：fd + 最后活动时间
{
    int fd;
    time_t last_active;     // 最后一次发送消息的时间
};

inline bool ret_check_exit(int ret, const string& msg = "", bool should_exit = true)
{ 
    if(ret < 0){
        char message[256];

        snprintf(message, sizeof(message), 
                 "%s | %d | %s | error : %s\n | errno = %d(%s) \n", 
                 (const char*)__FILE__, __LINE__, __FUNCTION__, msg.c_str(), errno, std::strerror(errno));
        perror(message);

        if(should_exit)
        {
            exit(EXIT_FAILURE);
        }
        return false;
    } 
    return true;
}

inline void ret_check_close(int ret, int fd, const string& msg = "", bool should_close = true)
{ 
    if(ret < 0){
        char message[256];

        snprintf(message, sizeof(message), 
                 "%s | %d | %s | error : %s\n | errno = %d(%s) \n", 
                 (const char*)__FILE__, __LINE__, __FUNCTION__, msg.c_str(), errno, std::strerror(errno));
        perror(message);

        if(should_close)
        {
            close(fd);
        }
    } 
}

inline void ret_check(int ret, const string& msg = "")
{ 
    if(ret < 0){
        char message[256];

        snprintf(message, sizeof(message), 
                 "%s | %d | %s | error : %s\n | errno = %d(%s) \n", 
                 (const char*)__FILE__, __LINE__, __FUNCTION__, msg.c_str(), errno, std::strerror(errno));
        perror(message);
    } 
}

bool test(){
    int ret = 0;

    // 1. 创建监听套接字
    int listen_fd = socket(AF_INET, SOCK_STREAM, 0);
    ret_check_exit(listen_fd, "socket failed");

    // 设置地址复用（避免TIME_WAIT 状态导致绑定失败）
    int opt = 1;
    // 套接字秒附属 | 选项所属的协议层 | 具体的选项名称 | 指向选项值的指针 | 选项值的长度
    // SOL_SOCKET 是通用套接字选项（最常用）,影响套接字本身的行为，与具体协议无关
    setsockopt(listen_fd, SOL_SOCKET, SO_REUSEADDR, &opt, sizeof(opt));

    // 2. 准备地址结构 & 绑定地址
    struct sockaddr_in server_addr = {0};
    server_addr.sin_family = AF_INET;
    //server_addr.sin_addr.s_addr = inet_addr("127.0.0.1");   // 监听特定IP
    server_addr.sin_addr.s_addr = INADDR_ANY;                 // 监听所有接口（推荐）
    server_addr.sin_port = htons(8080);

    // 注意结构体名称，是sock_addr*而不是sockaddr
    ret = bind(listen_fd, (struct sockaddr*)&server_addr, sizeof(server_addr));
    ret_check_exit(ret, "bind failed");

    // 3. 监听连接
    ret = listen(listen_fd, 5);
    ret_check_exit(ret, "listen failed");

    printf("Server listening on port 8080...\n");

    // 4. 创建epoll实例（在while外调用一次）
    int epoll_fd = epoll_create(1);
    ret_check_exit(epoll_fd, "epoll_create failed");

    // 5. 将监听套接字listen_fd加入epoll_fd监控（监听读事件）
    epoll_event ev;
    ev.events = EPOLLIN | EPOLLET;    // 监听读事件, 设置边缘模式
    ev.data.fd = listen_fd;
    ret = epoll_ctl(epoll_fd, EPOLL_CTL_ADD, listen_fd, &ev);
    ret_check_exit(ret, "epoll_ctl");

    vector<epoll_event> events(MAX_EVENTS); // 预分配空间存储就绪事件
    //vector<int> clients(MAX_EVENTS);      // 存储所有客户端连接的文件描述符
    vector<Client> clients;                 // 将存储数据类型更改为含有时间的结构体; 去掉初始大小，改为空向量vector<Client> clients，避免初始无效id
    
    char buffer[BUFFER_SIZE] = {0};

    while(true)
    {
        //int nfds = epoll_wait(epoll_fd, events.data(), MAX_EVENTS,  30 * 1000); // 获得就绪事件
        int nfds = epoll_wait(epoll_fd, events.data(), MAX_EVENTS, 1000);         // 此处的时间需要每秒检测1次，而非每30秒检测1次
        if(nfds > 0)
        {
            // 处理数据
            for(int i = 0; i < nfds; ++i){
                // 处理新连接
                if(events[i].data.fd == listen_fd)
                {
                    struct sockaddr_in client_addr;
                    socklen_t client_len = sizeof(client_addr);
                    int client_fd = accept(listen_fd, (struct sockaddr*)&client_addr, &client_len);
                    if(client_fd == -1)
                    {
                        perror("accept");
                        continue;
                    }

                    // 如果未设置客户端套接字为非阻塞模式，可能导致在边缘触发（ET）模式下阻塞
                    int flags = fcntl(client_fd, F_GETFL, 0);       // 获取当前文件描述符的标志
                    fcntl(client_fd, F_SETFL, flags | O_NONBLOCK);  // 在原有基础上，添加O_NONBLOCK（非阻塞模式）

                    // 处理客户端连接
                    //clients.push_back(client_fd);                 // 添加到客户端列表
                    clients.push_back({client_fd, time(nullptr)});  // 添加到客户端列表，初始化活动时间为当前时间
                    printf("Client%d connected: %s:%d\n", client_fd, inet_ntoa(client_addr.sin_addr), ntohs(client_addr.sin_port));

                    // 将新客户端套接字加入epoll监控（默认监听读事件）
                    ev.events = EPOLLIN | EPOLLET;  // 使用边缘触发模式（ET）
                    ev.data.fd = client_fd;
                    if(epoll_ctl(epoll_fd, EPOLL_CTL_ADD, client_fd, &ev) == -1)
                    {
                        perror("epoll_ctl: client_fd");
                        close(client_fd);
                    }
                }
                
                // 处理客户端数据可读事件
                else if(events[i].events & EPOLLIN)
                {
                    memset(buffer, 0, sizeof(buffer));
                    int client_fd = events[i].data.fd;
                    int n = read(client_fd, buffer, BUFFER_SIZE);

                    if (n <= 0) {
                        // 客户端关闭连接或发生错误
                        printf("Client disconnected\n");
                        close(client_fd);
                        epoll_ctl(epoll_fd, EPOLL_CTL_DEL, client_fd, nullptr);
                    } 
                    else 
                    {
                        printf("Received[Client:%d]: %s\n", client_fd, buffer);

                        // 动态调整事件：读完数据后，改为监听写事件（准备回显数据）
                        ev.events = EPOLLOUT | EPOLLET;
                        ev.data.fd = client_fd;
                        epoll_ctl(epoll_fd, EPOLL_CTL_MOD, client_fd, &ev);

                        // 收到客户端消息时，更新时间
                        auto it = find_if(clients.begin(), clients.end(), 
                                          [client_fd](const Client& c){
                                                return c.fd == client_fd;
                                          });
                        if(it != clients.end())
                        {
                            it->last_active = time(nullptr);    // 更新最后活动时间
                        }
                    }
                }
                
                // 处理客户端数据可写事件
                else if (events[i].events & EPOLLOUT) {
                    int client_fd = events[i].data.fd;
                    char response[BUFFER_SIZE];
                    snprintf(response, sizeof(response), "[client_fd:%d] %s", client_fd, buffer);
                    
                    int sender_fd = client_fd;
                    for(const auto& client : clients)
                    {
                        if(client.fd != sender_fd)
                        {
                            //write(client_fd, response, strlen(response));   // client_fd是当前客户端的意思
                            write(client.fd, response, strlen(response));   // client_fd是当前客户端的意思
                        }
                    }

                    // 【关键】写完后，改回监听读事件（等待下一次数据）
                    ev.events = EPOLLIN | EPOLLET;
                    ev.data.fd = client_fd;
                    epoll_ctl(epoll_fd, EPOLL_CTL_MOD, client_fd, &ev);
                }
            }
        }
        else if (nfds == 0)
        {
            // 注意，此处移除并不合适，因为如果就绪的时间小于wait设定的时间(s)，则不会进入分支判断
            #if 0
            // 连接关闭，关闭 fd 并从 epoll 中移除
            time_t now = time(nullptr);
            for(auto it = clients.begin(); it != clients.end();)
            {
                if(difftime(now, it->last_active) > 30)
                {
                    printf("Client %d timeout (30s), kicked out\n", it->fd);
                    // 从epoll中移除并关闭连接
                    epoll_ctl(epoll_fd, EPOLL_CTL_DEL, it->fd, nullptr);
                    close(it->fd);
                    // 从客户端列表移除
                    it = clients.erase(it);
                }
                else
                {
                    ++it;
                }
            }
            #endif

            //break;  // break会推出循环，群聊不能这么玩
        }
        else
        {
            if(errno == EAGAIN || errno == EWOULDBLOCK)
            {
                // 数据已读完，退出循环
                //break;    // 此处不应break，因为epoll_wait 不会返回 EAGAIN/EWOULDBLOCK
            }
            else
            {
                // 其他错误，处理错误并关闭 fd
                //break;    // 即使发生错误，也应继续运行而非退出
            }
        }
        
        // 超时踢出检查
        time_t now = time(nullptr);
        // 老师在这里使用了map键值对作为数据结构，set存储
        // 不过我想vector的话，注意++it的迭代器返回问题与移动问题就好了
        for(auto it = clients.begin(); it != clients.end();)
        {
            if(difftime(now, it->last_active) > 30)
            {
                printf("Client %d timeout (30s), kicked out\n", it->fd);
                // 从epoll中移除并关闭连接
                epoll_ctl(epoll_fd, EPOLL_CTL_DEL, it->fd, nullptr);
                close(it->fd);
                // 从客户端列表移除
                it = clients.erase(it);
            }
            else
            {
                ++it;
            }
        }
    }

    close(listen_fd);
    close(epoll_fd);
    return 0;
}

int main()
{
    test();
    return 0;
}

